square pattern in python language